package controller

import (
	"RocFamilySystem/internal/service"
	"RocFamilySystem/models"
	"RocFamilySystem/pkg/constant"
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"strconv"
	"strings"
)

type SysMenuController struct {
	BaseController
	SysMenuService *service.SysMenuService
}

// 初始化
func init() {
	registerContainer["SysMenuController"] = &SysMenuController{}
}

// SetService 设置服务
func (controller *SysMenuController) SetService(sysSysMenuService *service.SysMenuService) {
	controller.SysMenuService = sysSysMenuService
}

// RegisterRouter 注册路由
func (controller *SysMenuController) RegisterRouter(engine *gin.Engine) {
	prefix := "/sysMenu"
	g := engine.Group(prefix)
	prefixPermission := "permission" + strings.ReplaceAll(prefix, "/", ":")
	{
		//新增系统菜单
		g.POST("", controller.AddSysMenu)
		registerPermissionContainer("新增系统菜单", prefixPermission+":create", prefix, "POST")
		//删除系统菜单
		g.DELETE("/:id", controller.RemoveSysMenu)
		registerPermissionContainer("ID删除系统菜单", prefixPermission+":delete", prefix, "DELETE")
		g.DELETE("/batchDelete", controller.RemoveSysMenuBatch)
		registerPermissionContainer("ID删除系统菜单", prefixPermission+":batchDelete", prefix, "DELETE")
		//修改系统菜单
		g.PUT("", controller.ModifySysMenu)
		registerPermissionContainer("修改系统菜单", prefixPermission+":update", prefix, "PUT")
		//获取系统菜单列表
		g.POST("/list", controller.QuerySysMenu)
		registerPermissionContainer("查询系统菜单列表", prefixPermission+":list", prefix, "POST")
		//通过ID查询系统菜单
		g.GET("/:id", controller.QuerySysMenuById)
		registerPermissionContainer("ID查询系统菜单", prefixPermission+":id", prefix, "GET")
	}
}

// AddSysMenu 添加系统菜单
// @Summary AddSysMenu 添加系统菜单
// @Tags 系统菜单
// @Produce json
// @Param sysSysMenu body models.SysMenu true "系统菜单"
// @Success 200 {object} Result "成功"
// @Router /sysMenu [post]
func (controller *SysMenuController) AddSysMenu(ctx *gin.Context) {
	var sysSysMenu models.SysMenu
	err := ctx.ShouldBind(&sysSysMenu)
	controller.HandleError(err, constant.PARAMS_ERROR)
	controller.SysMenuService.AddSysMenu(&sysSysMenu)
	controller.CustomResult(ctx, 200, "添加系统菜单成功", nil)
}

// RemoveSysMenu 删除系统菜单
// @Summary RemoveSysMenu  删除系统菜单
// @Tags 系统菜单
// @Produce json
// @Param id path int true "系统菜单id"
// @Success 200 {object} Result "成功"
// @Router /sysMenu/:id [delete]
func (controller *SysMenuController) RemoveSysMenu(ctx *gin.Context) {
	sysSysMenuId, err := strconv.Atoi(ctx.Param("id"))
	controller.HandleError(err, constant.PARAMS_ERROR)
	controller.SysMenuService.RemoveSysMenu(sysSysMenuId)
	controller.CustomResult(ctx, 200, fmt.Sprintf("删除【%d】成功", sysSysMenuId), nil)
}

// RemoveSysMenuBatch 批量删除系统菜单
// @Summary RemoveSysMenu  批量删除系统菜单
// @Tags 系统缓存
// @Produce json
// @Param ids[] query []int true "系统缓存id"
// @Success 200 {object} Result "成功"
// @Router /sysMenu/batchDelete [delete]
func (controller *SysMenuController) RemoveSysMenuBatch(ctx *gin.Context) {
	ids := ctx.QueryArray("ids[]")
	if len(ids) == 0 {
		controller.HandleError(errors.New("参数缺失"), constant.PARAMS_ERROR)
	}
	controller.SysMenuService.RemoveSysMenuBatch(ids)
	controller.CustomResult(ctx, 200, fmt.Sprintf("删除【%v】成功", ids), nil)
}

// ModifySysMenu 修改系统菜单
// @Summary ModifySysMenu 修改系统菜单
// @Tags 系统菜单
// @Produce json
// @Param sysSysMenu body models.SysMenu true "系统菜单"
// @Success 200 {object} Result "成功"
// @Router /sysMenu [put]
func (controller *SysMenuController) ModifySysMenu(ctx *gin.Context) {
	var sysSysMenu models.SysMenu
	err := ctx.ShouldBind(&sysSysMenu)
	controller.HandleError(err, constant.PARAMS_ERROR)
	controller.SysMenuService.ModifySysMenu(&sysSysMenu)
	controller.CustomResult(ctx, 200, fmt.Sprintf("修改【%d】成功", *sysSysMenu.Id), nil)
}

// QuerySysMenu 查询系统菜单列表
// @Summary QuerySysMenu 查询系统菜单列表
// @Tags 系统菜单
// @Produce json
// @Param sysSysMenu body models.SysMenu true "系统菜单"
// @Success 200 {object} Result "成功"
// @Router /sysMenu/list [post]
func (controller *SysMenuController) QuerySysMenu(ctx *gin.Context) {
	pageNum, pageSize := controller.GetPageInfo(ctx)
	var sysSysMenu models.SysMenu
	err := ctx.ShouldBind(&sysSysMenu)
	controller.HandleError(err, constant.PARAMS_ERROR)
	data := controller.SysMenuService.QuerySysMenu(&sysSysMenu, pageNum, pageSize)
	controller.SuccessResult(ctx, data)
}

// QuerySysMenuById 查询系统菜单通过ID
// @Summary QuerySysMenuById  查询系统菜单通过ID
// @Tags 系统菜单
// @Produce json
// @Param id path int true "系统菜单id"
// @Success 200 {object} Result "成功"
// @Router /sysMenu/:id [get]
func (controller *SysMenuController) QuerySysMenuById(ctx *gin.Context) {
	sysSysMenuId, err := strconv.Atoi(ctx.Param("id"))
	controller.HandleError(err, constant.PARAMS_ERROR)
	data := controller.SysMenuService.QuerySysMenuById(sysSysMenuId)
	controller.SuccessResult(ctx, data)
}
